#pragma once

#include <iostream>
#include <string>
#include <fstream>
#include <atomic>
#include <boost/algorithm/string.hpp>

#include <sys/types.h>
#include <sys/stat.h>
#include <unistd.h>
#include <sys/time.h>

namespace ns_util
{
    const std::string temp_path="./temp/";

    class TimeUtil
    {
    public:

        // 获取秒级时间戳,以字符串形式返回
        static std::string GetTimeStamp()
        {
            struct timeval rtv;
            gettimeofday(&rtv,nullptr);

            return std::to_string(rtv.tv_sec);
        }

        static std::string GetTimeStampMs()
        {
            struct timeval rtv;
            gettimeofday(&rtv,nullptr);

            return std::to_string(rtv.tv_sec*1000 + rtv.tv_usec/1000);
        }
    };

    class PathUtil
    {
    public:

        // xxx
        // ./temp/xxx.cpp
        // ./temp/xxx.exe
        // ./temp/xxx.stderr
        static std::string AddSuffix(const std::string &file_name,const std::string &suffix)
        {
            std::string path_name=temp_path;
            path_name+=file_name;
            path_name+=suffix;

            return path_name;
        }

        // 根据file_name构建完整路径的源文件+带.cpp后缀
        static std::string Src(const std::string &file_name)
        {
            return AddSuffix(file_name,".cpp");
        }

        // 根据file_name构建完整路径的可执行程序文件+带.exe后缀
        static std::string Exe(const std::string &file_name)
        {
            return AddSuffix(file_name,".exe");    
        }

        // 根据file_name构建完整路径的编译错误文件+带.compile_error后缀
        static std::string CompileError(const std::string &file_name)
        {
            return AddSuffix(file_name,".compile_error");
        }  

        // 根据file_name构建完整路径的标准输入文件+带.stdin后缀
        static std::string Stdin(const std::string &file_name)
        {
            return AddSuffix(file_name,".stdin");
        }

        // 根据file_name构建完整路径的标准输出文件+带.stdout后缀
        static std::string Stdout(const std::string &file_name)
        {
            return AddSuffix(file_name,".stdout");
        }

        // 根据file_name构建完整路径的标准错误文件+带.stderr后缀
        static std::string Stderr(const std::string &file_name)
        {
            return AddSuffix(file_name,".stderr");
        }  
    };

    class FileUtil
    {
    public:

        // 判断某文件是否存在
        static bool IsFileExist(const std::string &path_name)
        {
            struct stat rstat;
            if(stat(path_name.c_str(),&rstat)<0)
            {
                return false;
            }

            return true;
        }

        // 获取唯一文件名
        static std::string GetUniqueFileName()
        {
            static std::atomic<long long> lat;
            std::string ms=TimeUtil::GetTimeStampMs();

            return ms+"_"+std::to_string(lat++);
        }

        // 向指定文件写入数据
        static bool WriteFile(const std::string &target,const std::string &data)
        {
            std::ofstream out(target);
            if(!out.is_open())
            {
                return false;
            }
            out.write(data.c_str(),data.size());

            out.close();
            return true;            
        }   

        // 向指定文件读取数据
        static bool ReadFile(const std::string &target, std::string *content, bool keep = false)
        {
            std::ifstream in(target);
            if(!in.is_open())
            {
                return false;
            }

            std::string line;
            while(std::getline(in,line))
            {
                line+=( keep ? "\n" : "" );
                content->append(line);
            }

            in.close();
            return true;
        }

        static bool UnlinkFile(const std::string &fname)
        {
            return 0==unlink(fname.c_str());
        }

    };

    class StringUtil
    {
    public:
        static void SplitString(const std::string &str,std::vector<std::string> *tokens,const std::string &sep)
        {
            // 使用boost库进行字符串分割
            boost::split((*tokens),str,boost::is_any_of(sep),boost::algorithm::token_compress_on);
        }

    };

}